from pydantic_settings import BaseSettings, SettingsConfigDict

class Settings(BaseSettings):
    # 数据库
    DB_HOST: str = "localhost"
    DB_PORT: int = 3306
    DB_USER: str = "root"
    DB_PASSWORD: str = "1234qwer#"
    DB_NAME: str = "shixun"
    # JWT
    JWT_SECRET_KEY: str = "secret-key"
    JWT_ALGORITHM: str = "HS256"
    ACCESS_TOKEN_EXPIRE_MINUTES: int = 24 * 60
    # 模型，还没设置
    MODEL_PATH: str = ""
    # 训练集位置
    DATASET_PATH: str = "D:\Traning_Dataset.xlsx"

    model_config = SettingsConfigDict(env_file=".env", extra="ignore")

settings = Settings()